在DAY18的時候有用過,可以對圖片做出一些處理,而這些處理也會影響到最後訓練的結果,
Transforms
網路上薩摩耶的圖片(圖片來源:https://www.dcard.tw/f/pet/p/233192736)
size = 100
transform = transforms.Resize(size)
new_img = transform(img)
new_img
上面是原圖,下面是resize之後的,可以看到比例明顯小了不少。
2. CenterCrop從最中間取一定範圍
size = 300
transform = transforms.Compose([
transforms.CenterCrop(size),
])
img1 = transform(img)
img1
3. Pad增加邊框
padding = 15
transform = transforms.Compose([
transforms.Resize((100,150)),
transforms.Pad(padding, fill=(230,200,250),padding_mode="constant"),
])
img1 = transform(img)
img1
resize之後加上邊框,還可以透過RGB調整顏色
4. RandomHorizontalFlip水平轉向
transform = transforms.Compose([
transforms.Resize((100,150)),
transforms.RandomHorizontalFlip(p=0.9),
])
img1 = transform(img)
img1
5. RandomHorizontalFlip垂直轉向
transform = transforms.Compose([
transforms.Resize((100,150)),
transforms.RandomVerticalFlip(p=0.9),
])
img1 = transform(img)
img1